# employment/schemas.py
from pydantic import BaseModel
from typing import Optional


class EmploymentBase(BaseModel):
    student_id: str
    student_name: Optional[str] = None
    class_name: Optional[str] = None
    job_search_start: Optional[str] = None
    offer_date: Optional[str] = None
    company_name: Optional[str] = None
    salary: Optional[int] = None


class EmploymentCreate(EmploymentBase):
    pass


class EmploymentUpdate(EmploymentBase):
    pass


class Employment(EmploymentBase):
    class Config:
        from_attributes = True